#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
将CSV文件中的内容写回到txt文件
"""

import csv
import os

def write_back_to_txt_files(csv_file, base_dir='01'):
    """
    从CSV文件读取内容并写回到对应的txt文件中
    
    参数:
        csv_file: CSV文件路径
        base_dir: txt文件所在的基础目录
    """
    print(f"开始处理 {csv_file} 中的内容...")
    
    with open(csv_file, 'r', encoding='utf-8') as f:
        reader = csv.reader(f)
        # 跳过标题行
        next(reader, None)
        
        total_count = 0
        success_count = 0
        
        for row in reader:
            if len(row) < 7:  # 确保行包含足够的列
                print(f"警告: 行 {row} 数据不完整，跳过")
                continue
                
            filename = row[0]
            # 将前5个话术合并到一个文件中，每个话术一行
            phrases = row[1:7]  # 包括原始内容和4个生成的话术
            
            # 过滤掉空字符串
            phrases = [phrase for phrase in phrases if phrase.strip()]
            
            if not phrases:
                print(f"警告: {filename} 没有有效的话术，跳过")
                continue
                
            # 构建完整的文件路径
            file_path = os.path.join(base_dir, filename)
            
            try:
                with open(file_path, 'w', encoding='utf-8') as txt_file:
                    # 每个话术写一行
                    txt_file.write('\n'.join(phrases))
                success_count += 1
                print(f"已更新: {file_path}")
            except Exception as e:
                print(f"错误: 更新 {file_path} 失败: {str(e)}")
            
            total_count += 1
        
        print(f"处理完成! 总共处理 {total_count} 个文件，成功更新 {success_count} 个文件。")

if __name__ == "__main__":
    # 让用户指定CSV文件和输出目录
    csv_file = input("请输入CSV文件路径（默认为'sales_scripts_with_similar.csv'）: ") or 'sales_scripts_with_similar.csv'
    base_dir = input("请输入txt文件所在的文件夹路径（默认为'01'）: ") or '01'
    
    write_back_to_txt_files(csv_file, base_dir) 